package com.tianji.aigc.config;

import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatModel;
import com.tianji.aigc.memory.RedisChatMemory;
import com.tianji.aigc.properties.SessionProperties;
import com.tianji.aigc.tools.CourseTool;
import com.tianji.aigc.tools.OrderTool;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * SpringAI配置类
 *
 * @Author mr.wu
 * @Date 2025-8-17 15:01
 */
@Configuration
public class SpringAiConfig {

    @Autowired
    private SessionProperties sessionProperties;

    @Bean
    public ChatClient chatClient(DashScopeChatModel chatModel, MessageChatMemoryAdvisor chatMemoryAdvisor, CourseTool courseTool, OrderTool orderTool) {
        return ChatClient.builder(chatModel) //对话模型对象
                .defaultSystem(sessionProperties.getSystemPrompt()) //添加提示词，定义角色信息
                .defaultAdvisors(new SimpleLoggerAdvisor(),//日志记录器
                        chatMemoryAdvisor) //添加保存会话记忆的功能
                .defaultTools(courseTool//注册Tool
                        , orderTool//预下单工具
                ).build();
    }

    /**
     * 声明一个bean,用来配置会话信息保存配置
     */
    @Bean
    public MessageChatMemoryAdvisor chatMemoryAdvisor(RedisChatMemory chatMemory) {
        return new MessageChatMemoryAdvisor(chatMemory);
    }
}